tensor([[[0.3800, 0.3800],
         [0.3800, 0.3800]],

        [[0.3100, 0.3100],
         [0.3100, 0.3100]],

        [[0.3100, 0.3100],
         [0.3100, 0.3100]]], grad_fn=<RoundBackward1>)